package com.xjaq.dify.common;


import lombok.Data;

import java.util.Map;

/**
 * 封装通用返回类
 */
@Data
public class Result {
    //定义两个常量，成功的code是200，失败的是-1
    public static final String SUCCESS_CODE = "200";
    public static final String ERROR_CODE = "-1";

    private String code;//code：接口的响应结果
    private Object data;//data：数据
    private String msg;//msg：存放错误信息
    private Map<String, Object> extra;//extra：存放额外的数据
    //无参数的成功方法，只返回成功代码“200”
    public static Result success() {
        Result result = new Result();
        result.setCode(SUCCESS_CODE);
        result.setMsg("success!");
        return result;
    }

    //有参数的成功方法，返回成功代码“200” 和 数据data
    public static Result success(Object data) {
        Result result = new Result();
        result.setCode(SUCCESS_CODE);
        result.setData(data);
        return result;
    }

    //失败的方法，返回自定义错误信息 和 错误代码“-1”
    public static Result error(String msg) {
        Result result = new Result();
        result.setCode(ERROR_CODE);
        result.setMsg(msg);
        return result;
    }

    //失败的方法，返回自定义错误信息 和 自定义错误代码
    public static Result error(String code, String msg) {
        Result result = new Result();
        result.setCode(code);
        result.setMsg(msg);
        return result;
    }
    //添加额外的数据
    public Result put(String key, Object value) {
        if (extra == null) {
            extra = new java.util.HashMap<>();
        }
        extra.put(key, value);
        return this;
    }
    //获取额外的数据
    public Object get(String key) {
        if (extra == null) {
            return null;
        }
        return extra.get(key);
    }
}
